#!/usr/bin/env python
# coding:utf-8

import requests
from hashlib import md5


class Chaojiying_Client(object):
    cjy_username = 'htg123'  # cjy 用户名
    cjy_password = 'htgcjy123'  # cjy 密码
    cjy_soft_id = '902457'   # cyj 生成的软件id
    cjy_yzm_type = '9004'    # 需要验证的验证码类型
    cyj_base_params = {
        'user': cjy_username,
        'pass2': md5(cjy_password.encode('utf8')).hexdigest(),
        'softid': cjy_soft_id,
    }

    cjy_headers = {
            'Connection': 'Keep-Alive',
            'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
    }

    def PostPic(self, im, codetype):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
        }
        params.update(self.cyj_base_params)
        files = {'userfile': ('ccc.jpg', im)}
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files, headers=self.cjy_headers)
        return r.json()

    def ReportError(self, im_id):
        """
        im_id:报错题目的图片ID
        """
        params = {
            'id': im_id,
        }
        params.update(self.cyj_base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.cjy_headers)
        return r.json()

    def cjy_run(self, img_path):
        im = open(img_path, 'rb').read()
        result_dict = self.PostPic(im, self.cjy_yzm_type)

        return result_dict

#
# if __name__ == '__main__':
#     # chaojiying = Chaojiying_Client('htg123', 'htgcjy123', '902457')	#用户中心>>软件ID 生成一个替换 96001
# 	# im = open('12306-1-2', 'rb').read()													#本地图片文件路径 来替换 a.jpg 有时WIN系统须要//
# 	# print (chaojiying.PostPic(im, 9004))										#1902 验证码类型  官方网站>>价格体系 3.4+版 print 后要加()
#     # {'err_no': 0, 'err_str': 'OK', 'pic_id': '2087522573328500002', 'pic_str': '109,80|271,56|126,145|276,155', 'md5': '97e003d814836d3e8b492d620b8fb56e'}
#     cjy = Chaojiying_Client()
#     cjy.cjy_run(img_path='../login_picture/yz_20191201_19.18.06.png')
